package algorithm;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

/**
 * 给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。
 *
 * 示例 1：
 * 输入：nums = [2,7,11,15], target = 9
 * 输出：[0,1]
 * 解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。
 *
 * 示例 2：
 * 输入：nums = [3,2,4], target = 6
 * 输出：[1,2]
 */
public class TwoSum {

    // 时间复杂度为n^2
    public static int[] twoSum(int[] nums, int target) {
        for(int i=0; i<nums.length; i++){
            for(int j=i+1; j<nums.length; j++){
                if(nums[i]+nums[j]==target){
                    return new int[]{i,j};
                }
            }
        }
        return new int[]{-1,-1};

    }

    // 时间复杂度为n
    public static int[] twoSum2(int[] nums, int target) {
        Map map = new HashMap<Integer,Integer>();
        for(int i=0; i<nums.length; i++){
            int tmp = target - nums[i];
            if(map.containsKey(tmp)){
                return new int[]{(Integer)map.get(tmp),i};
            }
            map.put(nums[i], i);
        }
        return new int[]{-1,-1};

    }

    // 如果输入一个递增的数组
    public static int[] twoSum3(int[] nums, int target) {
        int left = 0, right = nums.length-1;
        while(left<right){
            if(nums[left]+nums[right]==target){
                return new int[]{left,right};
            }
            if(nums[left]+nums[right]>target){
                right--;
            }
            if(nums[left]+nums[right]<target){
                left++;
            }
        }

        return new int[]{-1,-1};
    }

    public static void main(String[] args) {
        int[] result = twoSum3(new int[]{2,7,11,15}, 16);
        System.out.println(Arrays.toString(result)); // [1, 2]
    }

}
